import os
import sys
from grpc_tools import protoc

def generate_proto():
    # 确保目录存在
    # os.makedirs('generated', exist_ok=True)
    
    # 添加 dingding.proto 的生成
    protoc.main((
        '',
        '-I./',
        '--python_out=../generated',
        '--grpc_python_out=../generated',
        './dingding.proto',
    ))
    
    # 创建 __init__.py 文件
    # init_files = [
    #     'generated/__init__.py',
    #     'services/__init__.py',
    #     '__init__.py'
    # ]
    #
    # for init_file in init_files:
    #     with open(init_file, 'a'):
    #         pass
            
if __name__ == '__main__':
    generate_proto() 